def login():
    """登录函数"""
    username = input('请输入用户名')
    password = input('请输入密码')
    code = int(input('请输入验证码'))
    if username == 'admin' and password == '123456':
        if code == 8888:
            return '登录成功'
        else:
            return '验证码错误'
    else:
        return '用户名或密码错误'


if __name__ == '__main__':
    print(login())

